import onnx

# 加载模型文件
model = onnx.load(r"C:\Users\86198\Desktop\Wu\model\yolov5-6.0\yolov5-6.0\runs\train\exp33\exp33\weights\best.onnx")

# 修改图输入张量名称中的点
# 修改图输出张量名称中的点
for node in model.graph.node:
    node.name = node.name.replace('.', '_')
    print(output.name)
# 保存修改后的模型
onnx.save(model, "modified_model.onnx")
print("finish")